import asyncio

from bot.insurance_planner_gpt.agent.base_agent import LLMAgent

default_template = """
任务：把下面一段说的话按不同意群拆分，不同意群的句子中间用两个换行符拼接。
注意事项：
1.最多拆分成三句话
2.去除末尾的结束语

例子：
```
输入：
哎呀，抱歉之前没了解到您的情况~请问您是说金条账户暂时无法使用吗~如果您有财务需求，我们可以为您推荐申请京东联名信用卡或其他信贷产品~同时，您可以关注金条账户的变化，一旦额度恢复，我们会第一时间通知您哦~ 是不是有其他方面的需求我可以帮您解决呢
输出：
哎呀，抱歉之前没了解到您的情况~请问您是说金条账户暂时无法使用吗~

如果您有财务需求，我们可以为您推荐申请京东联名信用卡或其他信贷产品~

同时，您可以关注金条账户的变化，一旦额度恢复，我们会第一时间通知您哦~ 
```
输入：{conversation_input}
输出：
"""


class SplitOutputAgent(LLMAgent):

    def __init__(self, conversation_input):
        self.prompt = default_template.format(conversation_input=conversation_input)
        super().__init__(self.prompt)


if __name__ == "__main__":
    conversation_input = """
   一天又开始了，希望今天也是元气满满的一天哦！😄 我是您的专属保险规划师，随时准备为您提供帮助。请问有什么保险相关的问题需要我为您解答吗
     """
    splitOutputAgent = SplitOutputAgent(conversation_input)
    result = asyncio.run(splitOutputAgent.achat_auto_llm(type="gpt"))
    print(result)
